from .SelectLatentSize import SLS_CLASS_MAPPINGS, SLS_NAME_MAPPINGS
from .SelectLatentSizePlus import SLSP_CLASS_MAPPINGS, SLSP_NAME_MAPPINGS
from .TextEncode3in1 import TE3_CLASS_MAPPINGS, TE3_NAME_MAPPINGS
from .SamplerWithVae import SWV_CLASS_MAPPINGS, SWV_NAME_MAPPINGS
from .UpscaleByN import UBN_CLASS_MAPPINGS, UBN_NAME_MAPPINGS

# Register the node with ComfyUI.
# This is a dictionary that maps node names to their respective classes.
NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

NODE_CLASS_MAPPINGS.update(SLS_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(SLS_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(SLSP_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(SLSP_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(TE3_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(TE3_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(SWV_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(SWV_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(UBN_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(UBN_NAME_MAPPINGS)

WEB_DIRECTORY = "./js"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]